package cn.tedu.sp11.filter;


import cn.tedu.web.util.JsonResult;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.apache.commons.lang3.StringUtils;
import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

@Component
public class AccessFilter extends ZuulFilter {

    // 过滤器类型：包含pre, post, routing, error
    @Override
    public String filterType() {
        // return "pre";  // 上下二者等价 都是前置过滤器，字符串类型
        return FilterConstants.PRE_TYPE;
    }

    // 过滤器插入的位置（顺序号）
    // 在第五个过滤器上下文对象添加了serviceId
    // 在后面过滤器中才能访问serviceId
    @Override
    public int filterOrder() {
        return 6;
    }

    // 对用户请求判断，判断是否要执行过滤代码
    @Override
    public boolean shouldFilter() {
        /* 只对item-service 调用进行过滤
           如果调用user-service或者order-service则不执行过滤代码*/

        // 获取调用的服务id
        RequestContext ctx = RequestContext.getCurrentContext();
        String serviceId = (String) ctx.get(FilterConstants.SERVICE_ID_KEY);
        return "item-service".equalsIgnoreCase(serviceId);
    }

    // 过滤代码
    @Override
    public Object run() throws ZuulException {
        // 获得request对象
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();

        // 接受token参数
        String token = request.getParameter("token");

        // 如果没有token，组织访问，并像客户端返回响应
        if (StringUtils.isBlank(token)) {
            // 阻止继续访问
            ctx.setSendZuulResponse(false);
            // 向用户返回JsonResult{code:401, msg:not login, data:null}
            String json = JsonResult.err().code(JsonResult.NOT_LOGIN)
                    .msg("You are not login!").toString();
            ctx.addZuulResponseHeader("Content-Type", "application/json;charset=UTF-8");
            ctx.setResponseBody(json);
        }
        return null;  // 返回值没有任何作用
    }
}
